package com.zjs.vueMenu.service.impl;

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.zjs.vueMenu.entity.VueMenu;
import com.zjs.vueMenu.mapper.VueMenuMapper;
import com.zjs.vueMenu.service.VueMenuService;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

/**
 * <p>
 * 服务实现类
 * </p>
 *
 * @author zjs
 * @since 2020-07-06
 */
@Service
public class VueMenuServiceImpl extends ServiceImpl<VueMenuMapper, VueMenu> implements VueMenuService {

//    @Autowired
//    private VueMenuMapper vueMenuMapper;

    @Override
    public List<VueMenu> menuList() {
        List<VueMenu> vueMenus = this.list(null);

        List<VueMenu> pidMenu = vueMenus.stream().filter((e) -> e.getPid() == 0).collect(Collectors.toList());
        for (VueMenu menu : pidMenu) {
            List<VueMenu> items = new ArrayList<>();
            for (VueMenu vueMenu : vueMenus) {
                if (menu.getId().equals(vueMenu.getPid())) {
                    items.add(vueMenu);
                    menu.setItems(items);
                }
            }
        }
        return pidMenu;
    }
}
